import torch

a = torch.tensor([2, 2], dtype=torch.float32, device=torch.device('cpu'))